import torch

outputs = torch.randint(0, 10, (10, 3))
print(outputs)
box_label = torch.tensor([0, 1, 2, 2, 3, 3, 4, 5, 5, 5])
print([box_label != 5])
print(outputs[box_label != 5])
print(box_label[box_label != 5])
